package com.baiyun.lease.common.utils;


import com.baiyun.lease.common.exception.LeaseException;
import com.baiyun.lease.common.result.ResultCodeEnum;
import io.jsonwebtoken.*;
import io.jsonwebtoken.security.Keys;

import javax.crypto.SecretKey;
import java.util.Date;

/**
 * Created with IntelliJ IDEA.
 *
 * @Author: 许瑞淇
 * @Date: 2024/06/16/21:07
 * @Description:
 */
public class JwtUtil {

    //令牌过期时间设置为一小时
    private static long tokenExpiration = 60 * 60 * 1000L;
    //创建了一个用于对令牌进行签名的密钥，使用 HMAC-SHA256 算法。密钥是通过给定的字符串转换为字节数组后生成的。
    private static SecretKey tokenSignKey = Keys.hmacShaKeyFor("M0PKKI6pYGVWWfDZw90a0lTpGYX1d4AQ".getBytes());

    /**
     *         setSubject() 设置令牌的主题为 "USER_INFO"。
     *         setExpiration() 设置令牌过期时间（当前时间加上定义过期时间）
     *         claim() 向令牌添加用户ID和用户名的生命
     *         signWith(tokenSignKey, SignatureAlgorithm.HS256) 使用之前定义的密钥和 HMAC-SHA256 算法对令牌进行签名。
     *         compact() 生成最终的令牌字符串并返回
     */
    public static String createToken(Long userId, String username) {
        String token = Jwts.builder().
                setSubject("USER_INFO").
                setExpiration(new Date(System.currentTimeMillis() + tokenExpiration)).
                claim("userId", userId).
                claim("username", username).
                signWith(tokenSignKey, SignatureAlgorithm.HS256).
                compact();
        return token;
    }
    public static Claims parseToken(String token){
        if (token == null){
            throw new LeaseException(ResultCodeEnum.ADMIN_LOGIN_AUTH);
        }
        try {
            JwtParser jwtParser = Jwts.parserBuilder().setSigningKey(tokenSignKey).build();
            return jwtParser.parseClaimsJws(token).getBody();
        }
        catch (ExpiredJwtException e) {
            throw new LeaseException(ResultCodeEnum.TOKEN_EXPIRED);
        }
        catch (JwtException e) {
            throw new LeaseException(ResultCodeEnum.TOKEN_INVALID);
        }
    }
}
